import unittest
from 被测试代码 import *
class Test(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(1,2),3)
        self.assertNotEqual(add(1,2),4)
if __name__ == '__main__':
    unittest.main()